import requests

class TiebaSpider():
    def __init__(self,tieba_name):
        self.tieba_name = tieba_name
        self.url_temp = "https://tieba.baidu.com/f?kw=" + tieba_name + "&ie=utf-8&pn={}"
        self.headers = {"User-Agent":"Mozilla/5.0 (Windows NT 10.0; WOW64; Trident/7.0; rv:11.0) like Gecko"}

    def get_url_list(self): #构造url列表
        # url_list = []
        # for i in range(5):
        #     url_list.append(self.url_temp.format(i*50))
        # return url_list
        return [self.url_temp.format(i*50) for i in range(5)]

    def parse_url(self,url): #发送请求，获取响应
        print(url)
        response = requests.get(url,headers=self.headers)
        return response.content.decode()

    def save_file(self,html_str,page_num): #保存html字符串
        filepath = "{}-第{}页.html".format(self.tieba_name,page_num)
        with open(filepath,"w",encoding="utf-8") as f: #李毅吧-第4页
            f.write(html_str)

    def run(self): #实现主要逻辑
        #1.构造url列表
        url_list = self.get_url_list()
        #2.遍历，发送请求，获取响应
        for url in url_list:
            html_str = self.parse_url(url)
            #3.保存
            page_num = url_list.index(url) + 1 #页码数
            self.save_file(html_str,page_num)

if __name__ == '__main__':
    tieba_spider = TiebaSpider("李毅吧")
    tieba_spider.run()